package user_interface.dodaj_slowka.model;

import javax.swing.table.AbstractTableModel;
import java.util.ArrayList;

public class WordTableModel extends AbstractTableModel {
    private String[] tablica = {"Słowo po Polsku", "Słowo po angielsku"};

    ArrayList<Word> listaWierszy = new ArrayList<Word>();

    public void dodajSlowo(Word slowo) {
        listaWierszy.add(slowo);
        fireTableDataChanged();
    }

    public void usunSlowo(int numer){
        listaWierszy.remove(numer);
        fireTableDataChanged();
    }

    @Override
    public int getRowCount() {
        return listaWierszy.size();
    }

    @Override
    public int getColumnCount() {
        return tablica.length;
    }

    @Override
    public String getColumnName(int column) {
        return tablica[column];
    }

    @Override
    public Object getValueAt(int rowIndex, int columnIndex) {
        switch (columnIndex) {
            case 0:
                return listaWierszy.get(rowIndex).getSlowoPoPolsku();
            case 1:
                return listaWierszy.get(rowIndex).getSlowoPoAngielsku();
            default:
                return "";
        }
    }
}
